import argparse
from auto_optimizer import OnnxGraph

parser = argparse.ArgumentParser(description='infer shape.')
parser.add_argument('input_onnx', type=str, help='input onnx.')
parser.add_argument('output_onnx', type=str, help='output onnx.')
args = parser.parse_args()

g = OnnxGraph.parse(args.input_onnx)
g.infershape()
g.save(args.output_onnx)
